import requests
from lxml import etree
import os
#发送请求获取地址列表
headers={'user-agent':'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/127.0.0.0 Safari/537.36 Edg/127.0.0.0'}
resp_list=requests.get('https://yys.163.com/media/picture.html',headers=headers)
#获取所有背景的地址
e=etree.HTML(resp_list.text)
#取最后一个/前面所有相同内容+高清尺寸=横屏背景地址
#横屏
imgs1=[url[:url.rindex('/')]+'/2732×2048.jpg' for url in e.xpath('//div[@class="tab-cont"][1]/div/div/img/@data-src')]
#竖屏
imgs2=[url[:url.rindex('/')]+'/2732×2048.jpg' for url in e.xpath('//div[@class="tab-cont"][2]/div/div/img/@data-src')]

# 保存图片

#处理python文件系统
if not os.path.exists('heng'):
    os.makedirs('heng')
if not os.path.exists('shu'):
    os.makedirs('shu')

# resp=requests.get(图片的url（每个图片都不一样）)
for url in imgs1:   
    resp=requests.get(url,headers=headers)
    #给每个图片起名：日期+扩展名
    file_name=url[url.rindex('picture'):url.rindex('/')].replace('/','_')+'.jpg'
    
    print(f"正在下载"+file_name+"壁纸")

    with open(f'heng/{file_name}','wb')as f:
        f.write(resp.content)
